package com.chunfeng.mypen.config;

import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import com.chunfeng.mypen.utils.BaseContext;
import com.chunfeng.mypen.utils.SqlDateUtils;
import org.apache.ibatis.reflection.MetaObject;
import org.springframework.stereotype.Component;

/**
 * @Description: MyBatisPlus全局配置
 * @author: 小雷
 * @createTime: 2023-11-14 00:21:29
 */
@Component
public class MyMetaObjectHandler implements MetaObjectHandler {

    /**
     * 当前用户ID
     */
    private final String userId = BaseContext.getCurrentId();

    /**
     * 数据添加是: 自动填充策略
     *
     * @param metaObject 参数对象
     */
    @Override
    public void insertFill(MetaObject metaObject) {
        //生成ID值
        metaObject.setValue("createTime", SqlDateUtils.date);
        metaObject.setValue("createUser", userId == null ? "local" : userId);
    }

    /**
     * 数据修改时填充策略
     *
     * @param metaObject 参数对象
     */
    @Override
    public void updateFill(MetaObject metaObject) {
        metaObject.setValue("updateTime", SqlDateUtils.date);
        metaObject.setValue("updateUser", userId == null ? "local" : userId);
    }
}